"""
初始化Tongyi模型
安装必要的库
pip install langchain_community
pip install dashscope
pip install python-dotenv
创建.env文件，将Tongyi API Key添加到文件中
TONGYI_API_KEY=your_api_key_here
"""

from langchain_community.llms import Tongyi
from dotenv import load_dotenv
import os

# 加载环境变量
load_dotenv()

# 获取环境变量
api_key = os.getenv("DASHSCOPE_API_KEY")

# 初始化模型
llm = Tongyi(
    model="qwen-turbo",
    api_key=api_key,
    temperature=0.5,
    max_tokens=1024,
    max_retries=2
)

# 调用模型
res = llm.invoke("火锅最初是重庆码头的渔夫们弄出来的吗？")

print(res)
